package com.sfac.springboot.interceptor;

import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * @author jzq
 * @date 2022/8/22 18:11
 */
@SuppressWarnings({"all"})
@Component
public class ViewMappingInterceptor implements HandlerInterceptor {

    public final static Logger LOGGER = LoggerFactory.getLogger(ViewMappingInterceptor.class);

    @Override
    public boolean preHandle(HttpServletRequest request,
                             HttpServletResponse response,
                             Object handler) throws Exception {
        LOGGER.debug("==== Pre interceptor ====");
        return true;
    }

    @Override
    public void postHandle(HttpServletRequest request,
                           HttpServletResponse response,
                           Object handler,
                           ModelAndView modelAndView) throws Exception {
        LOGGER.debug("==== Post interceptor ====");

        // 判断请求是否为接口，如果为接口直接返回
        if (modelAndView == null) {
            return;
        }

        // 判断页面请求是否为重定向或者转发，如果是则返回
        if (modelAndView.getViewName().startsWith("redirect") ||
                modelAndView.getViewName().startsWith("forward")) {
            return;
        }

        // 获取请求路径
        String path = request.getServletPath();

        // 判断 modelmap 中是否有 template，有则跳过，无则根据请求路径设置 template
        String template = (String) modelAndView.getModelMap().get("template");
        if (StringUtils.isBlank(template)) {
            modelAndView.getModelMap().addAttribute("template",
                    path.substring(1, path.length()));
        }
        HandlerInterceptor.super.postHandle(request, response, handler, modelAndView);
    }

    @Override
    public void afterCompletion(HttpServletRequest request,
                                HttpServletResponse response,
                                Object handler,
                                Exception ex) throws Exception {
        LOGGER.debug("==== After interceptor ====");
    }
}
